import time

import datetime
from flask import current_app, jsonify
from flask import g
from flask import redirect
from flask import render_template
from flask import request
from flask import session
from flask import url_for

from info import constants
from info import user_login_data
from info.models import User, News, Category
from info.utils.image_storage import image_storage
from info.utils.response_code import RET
from . import admin_blue


##################################################################

# 功能: 管理员页面:管理员登陆页面
# 请求路径: /admin/login
# 请求方式: GET,
# 请求参数:GET,无, POST,username,password
# 返回值: GET渲染login.html页面, POST,login.html页面,errmsg
@admin_blue.route('/login', methods=['GET', 'POST'])
def admin_login():
    """
    -1.根据请求方式返回对应页面
    -2.获取参数
    -3.校验参数,为空校验
    -4.根据用户名,取出管理员对象,判断管理员对象是否存在
    -5.校验管理员密码是否匹配
    -6.记录管理员信息到 session 中
    -7.重定向到首页
    :return:
    """
    # -1.根据请求方式返回对应页面
    if request.method == "GET":
        if session.get("is_admin"):
            return redirect(url_for("admin.index"))
        return render_template("admin/login.html")

    # -2.获取参数
    username = request.form.get("username")
    password = request.form.get("password")

    # -3.校验参数,为空校验
    if not all([username, password]):
        return render_template("admin/login.html", errmsg="参数不完整")

    # -4.根据用户名,取出管理员对象,判断管理员对象是否存在
    try:
        admin = User.query.filter(User.mobile == username, User.is_admin == True).first()
    except Exception as e:
        current_app.logger.error(e)
        return render_template("admin/login.html", errmsg="管理员获取失败")

    if not admin:
        return render_template("admin/login.html", errmsg="管理员不存在")

    # -5.校验管理员密码是否匹配
    if not admin.check_password(password):
        return render_template("admin/login.html", errmsg="密码错误")

    # -6.记录管理员信息到 session 中
    session["user_id"] = admin.id
    session["nick_name"] = admin.nick_name
    session["mobile"] = admin.mobile
    session["is_admin"] = True

    # -7.重定向到首页
    # return redirect(url_for("admin.index"))
    return redirect(url_for("admin.index"))


# -------------------------------------------------------------------------
"""
1.因为现在使用的不是 ajax 提交了,所有使用的就不是 json 了,直接 返回页面即可
2.重定向到首页为何要使用 admin.index ?
    答:url_for()反解析,根据函数名,解析请求路径,因为 index 请求路径,默认是有个前缀的,前缀是 /admin,
    直接写 index 是找不到的,所以需要写 admin.
"""


# -------------------------------------------------------------------------

##################################################################

# 功能: 管理员页面:管理员首页
# 请求路径: /admin/index
# 请求方式: GET
# 请求参数: 无
# 返回值:渲染页面index.html,user字典数据
@admin_blue.route('/index')
@user_login_data
def index():
    # 携带数据渲染页面
    admin = g.user.to_dict() if g.user else ""
    return render_template("admin/index.html", admin=admin)


##################################################################

# 功能描述: 管理员页面:用户图表统计
# 请求路径: /admin/user_count
# 请求方式: GET
# 请求参数: 无
# 返回值:渲染页面user_count.html,字典数据
@admin_blue.route('/user_count')
def user_count():
    """
    -1.用户总数
    -2.用户每月活跃人数
    -3.用户每日活跃人数
    -4.对应时间段所活跃人数
    -5.拼接数据渲染页面
    :return:
    """
    # -1.用户总数
    try:
        total_count = User.query.filter(User.is_admin == False).count()
    except Exception as e:
        current_app.logger.error(e)
        return render_template("admin/user_count.html", data={"errmag": "用户总数查询失败"})

    # -2.用户每月活跃人数
    cal = time.localtime()
    try:
        # 本月的1号零点
        strat_mon_str = "%d-%d-01" % (cal.tm_year, cal.tm_mon)

        # 将本月一号零点,转成日期对象
        start_mon_data = datetime.datetime.strptime(strat_mon_str, "%Y-%m-%d")

        # 查询当前月活跃人数
        mon_count = User.query.filter(User.last_login >= start_mon_data, User.is_admin == False).count()
    except Exception as e:
        current_app.logger.error(e)
        return render_template("admin/user_count.html", data={"errmag": "用户查询当前月活跃人数查询失败"})

    # -3.用户每日活跃人数
    try:
        # 本日的零点
        strat_day_str = "%d-%d-%d" % (cal.tm_year, cal.tm_mon, cal.tm_mday)

        # 将本日零点,转成日期对象
        start_day_data = datetime.datetime.strptime(strat_day_str, "%Y-%m-%d")

        # 查询当天活跃人数
        day_count = User.query.filter(User.last_login >= start_day_data, User.is_admin == False).count()
    except Exception as e:
        current_app.logger.error(e)
        return render_template("admin/user_count.html", data={"errmag": "用户查询当天活跃人数查询失败"})

    # -4.对应时间段所活跃人数
    active_date = []
    active_count = []
    for i in range(0, 7):
        # 当天开始时间A
        begin_date = start_day_data - datetime.timedelta(days=i)

        # 当天开始时间的后一天时间B
        end_date = start_day_data - datetime.timedelta(days=i - 1)

        # 添加当天开始时间字符串到活跃日期中
        active_date.append(begin_date.strftime("%m-%d"))

        # 查询时间A到时间B这一天的注册人数
        everyday_active_count = User.query.filter(User.is_admin == False, User.last_login >= begin_date,
                                                  User.last_login <= end_date).count()

        # 添加当天注册人数到读取数量中
        active_count.append(everyday_active_count)

    # 为了方便查看图表,反转容器
    active_date.reverse()
    active_count.reverse()

    # -5.拼接数据渲染页面
    data = {
        "total_count": total_count,
        "mon_count": mon_count,
        "day_count": day_count,
        "active_date": active_date,
        "active_count": active_count
    }
    return render_template("admin/user_count.html", data=data)


##################################################################

# 功能描述:管理员页面:用户列表
# 请求路径: /admin/user_list
# 请求方式: GET
# 请求参数: p
# 返回值:渲染user_list.html页面,data字典数据
@admin_blue.route('/user_list')
def user_list():
    # 1.获取参数
    page = request.args.get("p", "1")

    # 2.参数类型转换
    try:
        page = int(page)
    except Exception as e:
        page = 1

    # 3.分页查询
    try:
        paginate = User.query.filter(User.is_admin == False).order_by(User.create_time.desc()).paginate(page, 10, False)
    except Exception as e:
        current_app.logger.error(e)
        return render_template("admin/user_list.html", data={"errmsg": "用户获取失败"})

    # 4.获取分页对象属性:总页数,当前页,当前页对象列表
    totalPage = paginate.pages
    currentPage = paginate.page
    items = paginate.items

    # 5.用户对象列表转换字典列表
    user_list = []
    for item in items:
        user_list.append(item.to_admin_dict())

    # 6.携带数据渲染页面
    data = {
        "totalPage": totalPage,
        "currentPage": currentPage,
        "user_list": user_list
    }
    return render_template("admin/user_list.html", data=data)


##################################################################

# 功能描述:获取/设置新闻审核
# 请求路径: /admin/news_review
# 请求方式: GET
# 请求参数: GET, p,keywords
# 返回值:渲染user_list.html页面,data字典数据
@admin_blue.route('/news_review')
def news_review():
    # 1.获取参数
    page = request.args.get("p", "1")
    keywords = request.args.get("keywords")

    # 2.参数类型转换
    try:
        page = int(page)
    except Exception as e:
        page = 1

    # 3.分页查询
    try:
        # 条件容器
        filters = [News.status != 0]
        # 判断是否输入了关键字搜索
        if keywords:
            filters.append(News.title.contains(keywords))

        paginate = News.query.filter(*filters).order_by(News.create_time.desc()).paginate(page, 10, False)
    except Exception as e:
        current_app.logger.error(e)
        return render_template("admin/news_review.html", data={"errmsg": "新闻获取失败"})

    # 4.获取分页属性
    totalPage = paginate.pages
    currentPage = paginate.page
    items = paginate.items

    # 5.新闻对象列表,转换字典列表
    news_list = []
    for item in items:
        news_list.append(item.to_review_dict())

    # 6.携带数据渲染页面
    data = {
        "totalPage": totalPage,
        "currentPage": currentPage,
        "news_list": news_list
    }

    return render_template("admin/news_review.html", data=data)


##################################################################

# 功能描述:获取/设置新闻审核详情
# 请求路径: /admin/news_review_detail
# 请求方式: GET,POST
# 请求参数: GET, news_id, POST,news_id, action
# 返回值:GET,渲染news_review_detail.html页面,data字典数据
@admin_blue.route('/news_review_detail', methods=['GET', 'POST'])
def news_review_detail():
    """
    -1.判断请求方式,如果是GET
    -2.根据编号,查询新闻
    -3.返回数据渲染页面
    -4.如果是 POST 获取请求参数
    -5.校验参数,为空校验
    -6.判断操作类型
    -7.根据新闻编号,取出新闻对象,并判断新闻是否存在
    -8.根据操作类型,修改新闻状态
    -9.返回响应
    :return:
    """
    # -1.判断请求方式,如果是GET
    if request.method == "GET":
        # 获取参数,为空校验
        news_id = request.args.get("news_id")

        if not news_id: return jsonify(errno=RET.PARAMERR, errmsg="新闻审核:参数不全")

        # -2.根据编号,查询新闻,为空校验
        try:
            news = News.query.get(news_id)
        except Exception as e:
            current_app.logger.error(e)
            return jsonify(errno=RET.DBERR, errmsg="查询新闻失败")

        if not news: return jsonify(errno=RET.NODATA, errmsg="新闻对象不存在")

        # -3.返回数据渲染页面
        return render_template("admin/news_review_detail.html", news=news.to_dict())

    # -4.如果是POST获取请求参数
    news_id = request.json.get("news_id")
    action = request.json.get("action")

    # -5.校验参数, 为空校验
    if not all([news_id, action]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数不全")

    # -6.判断操作类型
    if not action in ["accept", "reject"]:
        return jsonify(errno=RET.DATAERR, errmsg="操作类型有误")

    # -7.根据新闻编号, 取出新闻对象, 并判断新闻是否存在
    try:
        news = News.query.get(news_id)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="获取新闻失败")

    if not news: return jsonify(errno=RET.NODATA, errmsg="新闻对象不存在")

    # -8.根据操作类型, 修改新闻状态
    if action == "accept":
        news.status = 0
    else:
        # 拒绝原因
        news.reason = request.json.get("reason", "")
        news.status = -1
    # -9.返回响应
    return jsonify(errno=RET.OK, errmsg="操作成功")


##################################################################

# 功能描述:新闻版式编辑
# 请求路径: /admin/news_edit
# 请求方式: GET
# 请求参数: GET, p, keywords
# 返回值:GET,渲染news_edit.html页面,data字典数据
@admin_blue.route('/news_edit')
def news_edit():
    # 1.获取参数
    page = request.args.get("p", "1")
    keywords = request.args.get("keywords")

    # 2.参数类型转换
    try:
        page = int(page)
    except Exception as e:
        page = 1

    # 3.分页查询
    try:
        # 条件容器
        filters = []
        # 判断是否输入了关键字
        if keywords:
            filters.append(News.title.contains(keywords))

        paginate = News.query.filter(*filters).order_by(News.create_time.desc()).paginate(page, 5, False)
    except Exception as e:
        current_app.logger.error(e)
        return render_template("admin/news_edit.html", data={"errmsg": "新闻获取失败"})

    # 4.获取分页属性
    totalPage = paginate.pages
    currentPage = paginate.page
    items = paginate.items

    # 5.新闻对象列表,转换字典列表
    news_list = []
    for item in items:
        news_list.append(item.to_review_dict())

    # 6.携带数据渲染页面
    data = {
        "totalPage": totalPage,
        "currentPage": currentPage,
        "news_list": news_list
    }
    return render_template("admin/news_edit.html", data=data)


##################################################################

# 功能描述:获取/设置新闻版式编辑详情
# 请求路径: /admin/news_edit_detail
# 请求方式: GET, POST
# 请求参数: GET, news_id, POST(news_id,title,digest,content,index_image,category_id)
# 返回值:GET,渲染news_edit_detail.html页面,data字典数据, POST(errno,errmsg)
@admin_blue.route('/news_edit_detail', methods=['GET', 'POST'])
def news_edit_detail():
    """
    - 1.判断请求方式
        - 2.获取新闻编号
        - 3.根据新闻编号查询新闻对象,判断新闻对象是否存在
        - 4.查询所有分类,并转成字典列表
        - 5.返回数据
    :return:
    """
    # - 1.判断请求方式
    if request.method == "GET":

        # - 2.获取新闻编号
        news_id = request.args.get("news_id")
        if not news_id: return jsonify(errno=RET.PARAMERR, errmsg="参数不全")

        # - 3.根据新闻编号查询新闻对象,判断新闻对象是否存在
        try:
            news = News.query.get(news_id)
        except Exception as e:
            current_app.logger.error(e)
            return jsonify(errno=RET.DBERR, errmsg="获取新闻失败")

        if not news: return jsonify(errno=RET.NODATA, errmsg="新闻不存在")

        # - 4.查询所有分类,并转成字典列表
        try:
            categories = Category.query.all()
        except Exception as e:
            current_app.logger.error(e)
            return jsonify(errno=RET.DBERR, errmsg="查询分类失败")

        # - 5.新闻对象列表转成字典列表,返回数据
        categories_list = []
        for category in categories:
            categories_list.append(category.to_dict())

        return render_template("admin/news_edit_detail.html", news=news.to_dict(), categories=categories_list)

    # 如果是POST请求:
    # 1.获取参数:news_id,title,digest,content,index_image,category_id
    news_id = request.form.get("news_id")
    title = request.form.get("title")
    digest = request.form.get("digest")
    content = request.form.get("content")
    index_image = request.files.get("index_image")
    category_id = request.form.get("category_id")

    # 2.参数为空校验
    if not all([news_id, title, digest, content, index_image, category_id]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数不全")

    # 3.根据新闻编号取出新闻对象,并判断是否存在
    try:
        news = News.query.get(news_id)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="获取新闻失败")

    if not news: return jsonify(errno=RET.NODATA, errmsg="新闻不存在")

    # 4.上传图片至七牛云
    try:
        image_name = image_storage(index_image.read())
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.THIRDERR, errmsg="七牛云异常")

    if not image_name: return jsonify(errno=RET.NODATA, errmsg="上传图片失败")

    # 5.修改这个新闻在服务器的数据
    try:
        news.title = title
        news.digest = digest
        news.content = content
        news.category_id = category_id
        news.index_image_url = constants.QINIU_DOMIN_PREFIX + image_name
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="修改失败")

    # 6.返回响应
    return jsonify(errno=RET.OK, errmsg="新闻版式编辑:修改成功")


##################################################################

# 功能描述: 获取所有分类
# 请求路径: /admin/news_category
# 请求方式: GET
# 请求参数: GET,无
# 返回值:GET,渲染news_type.html页面, data数据
@admin_blue.route('/news_category')
def news_category():
    try:
        categories = Category.query.all()
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="查询分类失败")

    if not categories: return jsonify(errno=RET.NODATA, errmsg="获取新闻分类失败")

    category_list = []
    for category in categories:
        category_list.append(category.to_dict())

    return render_template("admin/news_type.html", categories=category_list)
